package net.sahv.bdyz.util;

import java.io.BufferedReader;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.FileInputStream;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.Iterator;
import java.util.Map;

public class FileUpLoadNew {


    /** 

     * 上传图片  王聪朝 跨域上传图片

     * @param urlStr 

     * @param textMap 

     * @param fileMap 
     * @param imgEndString 

     * @return 

     */  

    public static String formUpload(String urlStr, Map<String, String> textMap, Map<String, InputStream> fileMap, String imgEndString) {  

        String res = "";  

        HttpURLConnection conn = null;  

        String BOUNDARY = "---------------------------123821742118716"; //boundary就是request头和上传文件内容的分隔符    

        try {  

            URL url = new URL(urlStr);  

            conn = (HttpURLConnection) url.openConnection();  

            conn.setConnectTimeout(5000);  

            conn.setReadTimeout(30000);  

            conn.setDoOutput(true);  

            conn.setDoInput(true);  

            conn.setUseCaches(false);  

            conn.setRequestMethod("POST");  

            conn.setRequestProperty("Connection", "Keep-Alive");  

            conn.setRequestProperty("User-Agent", "Mozilla/5.0 (Windows; U; Windows NT 6.1; zh-CN; rv:1.9.2.6)");  

            conn.setRequestProperty("Content-Type", "multipart/form-data; boundary=" + BOUNDARY);  

  

            OutputStream out = new DataOutputStream(conn.getOutputStream());  

            // text    

            if (textMap != null) {  

                StringBuffer strBuf = new StringBuffer();  

                Iterator<Map.Entry<String, String>> iter = textMap.entrySet().iterator();  

                while (iter.hasNext()) {  

                    Map.Entry<String, String> entry = iter.next();  

                    String inputName = (String) entry.getKey();  

                    String inputValue = (String) entry.getValue();  

                    if (inputValue == null) {  

                        continue;  

                    }  

                    strBuf.append("\r\n").append("--").append(BOUNDARY).append("\r\n");  

                    strBuf.append("Content-Disposition: form-data; name=\"" + inputName + "\"\r\n\r\n");  

                    strBuf.append(inputValue);  

                }  

                out.write(strBuf.toString().getBytes());  

            }  

  

            // file    

            if (fileMap != null) {  

                Iterator<Map.Entry<String, InputStream>> iter = fileMap.entrySet().iterator();  

                while (iter.hasNext()) {  

                    Map.Entry<String, InputStream> entry = iter.next();  

                    String inputName = (String) entry.getKey();  

                   

                    FileInputStream inputValue =   (FileInputStream) entry.getValue();  

                    if (inputValue == null) {  

                        continue;  

                    }
                    
                    String filename = "bdyz_"+System.currentTimeMillis()+".png";
                    //如果文件后缀存在，则替换成上传的文件后缀名
                    if(imgEndString!=null&&!imgEndString.equals("")){
                    	filename = "bdyz_"+System.currentTimeMillis()+"."+imgEndString; 
                    }

                    String contentType = "image/png";  

                    StringBuffer strBuf = new StringBuffer();  

                    strBuf.append("\r\n").append("--").append(BOUNDARY).append("\r\n");  

                    strBuf.append("Content-Disposition: form-data; name=\"" + inputName + "\"; filename=\"" + filename + "\"\r\n");  

                    strBuf.append("Content-Type:" + contentType + "\r\n\r\n");  

  

                    out.write(strBuf.toString().getBytes());  

  

                    

                    DataInputStream in = new DataInputStream(inputValue);  

                    int bytes = 0;  

                    byte[] bufferOut = new byte[1024];  

                    while ((bytes = in.read(bufferOut)) != -1) {  

                        out.write(bufferOut, 0, bytes);  

                    }  

                    in.close();  

                }  

            }  

  

            byte[] endData = ("\r\n--" + BOUNDARY + "--\r\n").getBytes();  

            out.write(endData);  

            out.flush();  

            out.close();  

  

            // 读取返回数据    

            StringBuffer strBuf = new StringBuffer();  

            BufferedReader reader = new BufferedReader(new InputStreamReader(conn.getInputStream()));  

            String line = null;  

            while ((line = reader.readLine()) != null) {  

                strBuf.append(line).append("\n");  

            }  

            res = strBuf.toString();  

            reader.close();  

            reader = null;  

        } catch (Exception e) {  

            System.out.println("发送POST请求出错。" + urlStr);  

            e.printStackTrace();  

        } finally {  

            if (conn != null) {  

                conn.disconnect();  

                conn = null;  

            }  

        }  

        return res;  

    } 
}
